/* Copyright 2023 Dual Tachyon
 * https://github.com/DualTachyon
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *     Unless required by applicable law or agreed to in writing, software
 *     distributed under the License is distributed on an "AS IS" BASIS,
 *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *     See the License for the specific language governing permissions and
 *     limitations under the License.
 */

#ifdef ENABLE_FMRADIO
#include "app/fm.h"
#endif
#include "audio.h"
#include "bsp/dp32g030/gpio.h"
#ifdef ENABLE_FMRADIO
#include "driver/bk1080.h"
#endif
#include "driver/bk4819.h"
#include "driver/gpio.h"
#include "driver/system.h"
#include "driver/systick.h"
#include "functions.h"
#include "misc.h"
#include "settings.h"
#include "ui/ui.h"
#include "utils/music.h"

BEEP_Type_t gBeepToPlay = BEEP_NONE;


void AUDIO_meep(void)
{
  AUDIO_AudioPathOff();
  SYSTEM_DelayMs(20);

  uint16_t ToneConfig = BK4819_ReadRegister(BK4819_REG_71);
  AUDIO_AudioPathOn();

  //  for (uint16_t ToneFrequency = 220; ToneFrequency < 1500; ToneFrequency += 100)
  //    BK4819_PlaySingleTone(ToneFrequency, 60, 127, true);

  for (const uint8_t *musicp = musicdata; *musicp != 255; musicp += 4)
    {
      int freq = musicp[1] + (musicp[0] << 8); /* in Hz */
      int dur  = musicp[3] + (musicp[2] << 8); /* in 1/100s */
      dur *= 5;
      BK4819_PlaySingleTone(freq, dur, 64, true);
    }

  AUDIO_AudioPathOff();
  SYSTEM_DelayMs(5);
  BK4819_TurnsOffTones_TurnsOnRX();
  SYSTEM_DelayMs(5);
  BK4819_WriteRegister(BK4819_REG_71, ToneConfig);
}



void AUDIO_PlayBeep(BEEP_Type_t Beep)
{
  if (Beep != BEEP_880HZ_60MS_TRIPLE_BEEP &&
      Beep != BEEP_500HZ_60MS_DOUBLE_BEEP &&
      Beep != BEEP_440HZ_500MS &&
      Beep != BEEP_880HZ_200MS &&
      Beep != BEEP_880HZ_500MS &&
      !gEeprom.BEEP_CONTROL)
    return;

#ifdef ENABLE_AIRCOPY
  if (gScreenToDisplay == DISPLAY_AIRCOPY)
    return;
#endif

  if (gCurrentFunction == FUNCTION_RECEIVE)
    return;

  if (gCurrentFunction == FUNCTION_MONITOR)
    return;

#ifdef ENABLE_FMRADIO
  if (gFmRadioMode)
    BK1080_Mute(true);
#endif

  AUDIO_AudioPathOff();

  if (gCurrentFunction == FUNCTION_POWER_SAVE && gRxIdleMode)
    BK4819_RX_TurnOn();

  SYSTEM_DelayMs(20);

  uint16_t ToneConfig = BK4819_ReadRegister(BK4819_REG_71);

  uint16_t ToneFrequency;
  switch (Beep)
    {
    default:
    case BEEP_NONE:
      ToneFrequency = 220;
      break;
    case BEEP_1KHZ_60MS_OPTIONAL:
      ToneFrequency = 1000;
      break;
    case BEEP_500HZ_60MS_DOUBLE_BEEP_OPTIONAL:
    case BEEP_500HZ_60MS_DOUBLE_BEEP:
      ToneFrequency = 500;
      break;
    case BEEP_440HZ_40MS_OPTIONAL:
    case BEEP_440HZ_500MS:
      ToneFrequency = 440;
      break;
    case BEEP_880HZ_40MS_OPTIONAL:
    case BEEP_880HZ_60MS_TRIPLE_BEEP:
    case BEEP_880HZ_200MS:
    case BEEP_880HZ_500MS:
      ToneFrequency = 880;
      break;
    }

  BK4819_PlayTone(ToneFrequency, true);

  SYSTEM_DelayMs(2);

  AUDIO_AudioPathOn();

  SYSTEM_DelayMs(60);

  uint16_t Duration;
  switch (Beep)
    {
    case BEEP_880HZ_60MS_TRIPLE_BEEP:
      BK4819_ExitTxMute();
      SYSTEM_DelayMs(60);
      BK4819_EnterTxMute();
      SYSTEM_DelayMs(20);
      [[fallthrough]];
    case BEEP_500HZ_60MS_DOUBLE_BEEP_OPTIONAL:
    case BEEP_500HZ_60MS_DOUBLE_BEEP:
      BK4819_ExitTxMute();
      SYSTEM_DelayMs(60);
      BK4819_EnterTxMute();
      SYSTEM_DelayMs(20);
      [[fallthrough]];
    case BEEP_1KHZ_60MS_OPTIONAL:
      BK4819_ExitTxMute();
      Duration = 60;
      break;
    case BEEP_880HZ_40MS_OPTIONAL:
    case BEEP_440HZ_40MS_OPTIONAL:
      BK4819_ExitTxMute();
      Duration = 40;
      break;
    case BEEP_880HZ_200MS:
      BK4819_ExitTxMute();
      Duration = 200;
      break;
    case BEEP_440HZ_500MS:
    case BEEP_880HZ_500MS:
    default:
      BK4819_ExitTxMute();
      Duration = 500;
      break;
    }

  SYSTEM_DelayMs(Duration);
  BK4819_EnterTxMute();
  SYSTEM_DelayMs(20);

  AUDIO_AudioPathOff();

  SYSTEM_DelayMs(5);
  BK4819_TurnsOffTones_TurnsOnRX();
  SYSTEM_DelayMs(5);
  BK4819_WriteRegister(BK4819_REG_71, ToneConfig);

  if (gEnableSpeaker)
    AUDIO_AudioPathOn();

#ifdef ENABLE_FMRADIO
  if (gFmRadioMode)
    BK1080_Mute(false);
#endif

  if (gCurrentFunction == FUNCTION_POWER_SAVE && gRxIdleMode)
    BK4819_Sleep();

#ifdef ENABLE_VOX
  gVoxResumeCountdown = 80;
#endif

}

#ifdef ENABLE_VOICE

static const uint8_t VoiceClipLengthChinese[58] =
  {
    0x32, 0x32, 0x32, 0x37, 0x37, 0x32, 0x32, 0x32,
    0x32, 0x37, 0x37, 0x32, 0x64, 0x64, 0x64, 0x64,
    0x64, 0x69, 0x64, 0x69, 0x5A, 0x5F, 0x5F, 0x64,
    0x64, 0x69, 0x64, 0x64, 0x69, 0x69, 0x69, 0x64,
    0x64, 0x6E, 0x69, 0x5F, 0x64, 0x64, 0x64, 0x69,
    0x69, 0x69, 0x64, 0x69, 0x64, 0x64, 0x55, 0x5F,
    0x5A, 0x4B, 0x4B, 0x46, 0x46, 0x69, 0x64, 0x6E,
    0x5A, 0x64,
  };

static const uint8_t VoiceClipLengthEnglish[76] =
  {
    0x50, 0x32, 0x2D, 0x2D, 0x2D, 0x37, 0x37, 0x37,
    0x32, 0x32, 0x3C, 0x37, 0x46, 0x46, 0x4B, 0x82,
    0x82, 0x6E, 0x82, 0x46, 0x96, 0x64, 0x46, 0x6E,
    0x78, 0x6E, 0x87, 0x64, 0x96, 0x96, 0x46, 0x9B,
    0x91, 0x82, 0x82, 0x73, 0x78, 0x64, 0x82, 0x6E,
    0x78, 0x82, 0x87, 0x6E, 0x55, 0x78, 0x64, 0x69,
    0x9B, 0x5A, 0x50, 0x3C, 0x32, 0x55, 0x64, 0x64,
    0x50, 0x46, 0x46, 0x46, 0x4B, 0x4B, 0x50, 0x50,
    0x55, 0x4B, 0x4B, 0x32, 0x32, 0x32, 0x32, 0x37,
    0x41, 0x32, 0x3C, 0x37,
  };

VOICE_ID_t        gVoiceID[8];
uint8_t           gVoiceReadIndex;
uint8_t           gVoiceWriteIndex;
volatile uint16_t gCountdownToPlayNextVoice_10ms;
volatile bool     gFlagPlayQueuedVoice;
VOICE_ID_t        gAnotherVoiceID = VOICE_ID_INVALID;


static void AUDIO_PlayVoice(uint8_t VoiceID)
{
  unsigned int i;

  GPIO_SetBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);
  SYSTEM_DelayMs(20);
  GPIO_ClearBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);

  for (i = 0; i < 8; i++)
    {
      if ((VoiceID & 0x80U) == 0)
	GPIO_ClearBit(&GPIOA->DATA, GPIOA_PIN_VOICE_1);
      else
	GPIO_SetBit(&GPIOA->DATA, GPIOA_PIN_VOICE_1);

      SYSTICK_DelayUs(1000);
      GPIO_SetBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);
      SYSTICK_DelayUs(1200);
      GPIO_ClearBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);
      VoiceID <<= 1;
      SYSTICK_DelayUs(200);
    }
}

void AUDIO_PlaySingleVoice(bool bFlag)
{
  uint8_t VoiceID;
  uint8_t Delay;

  VoiceID = gVoiceID[0];

  if (gEeprom.VOICE_PROMPT != VOICE_PROMPT_OFF && gVoiceWriteIndex > 0)
    {
      if (gEeprom.VOICE_PROMPT == VOICE_PROMPT_CHINESE)
	{	// Chinese
	  if (VoiceID >= ARRAY_SIZE(VoiceClipLengthChinese))
	    goto Bailout;

	  Delay    = VoiceClipLengthChinese[VoiceID];
	  VoiceID += VOICE_ID_CHI_BASE;
	}
      else
	{	// English
	  if (VoiceID >= ARRAY_SIZE(VoiceClipLengthEnglish))
	    goto Bailout;

	  Delay    = VoiceClipLengthEnglish[VoiceID];
	  VoiceID += VOICE_ID_ENG_BASE;
	}

      if (FUNCTION_IsRx())   // 1of11
	BK4819_SetAF(BK4819_AF_MUTE);

#ifdef ENABLE_FMRADIO
      if (gFmRadioMode)
	BK1080_Mute(true);
#endif

      AUDIO_AudioPathOn();

#ifdef ENABLE_VOX
      gVoxResumeCountdown = 2000;
#endif

      SYSTEM_DelayMs(5);
      AUDIO_PlayVoice(VoiceID);

      if (gVoiceWriteIndex == 1)
	Delay += 3;

      if (bFlag)
	{
	  SYSTEM_DelayMs(Delay * 10);

	  if (FUNCTION_IsRx())	// 1of11
	    RADIO_SetModulation(gRxVfo->Modulation);

#ifdef ENABLE_FMRADIO
	  if (gFmRadioMode)
	    BK1080_Mute(false);
#endif

	  if (!gEnableSpeaker)
	    AUDIO_AudioPathOff();

	  gVoiceWriteIndex    = 0;
	  gVoiceReadIndex     = 0;

#ifdef ENABLE_VOX
	  gVoxResumeCountdown = 80;
#endif

	  return;
	}

      gVoiceReadIndex                = 1;
      gCountdownToPlayNextVoice_10ms = Delay;
      gFlagPlayQueuedVoice           = false;

      return;
    }

 Bailout:
  gVoiceReadIndex  = 0;
  gVoiceWriteIndex = 0;
}

void AUDIO_SetVoiceID(uint8_t Index, VOICE_ID_t VoiceID)
{
  if (Index >= ARRAY_SIZE(gVoiceID))
    return;

  if (Index == 0)
    {
      gVoiceWriteIndex = 0;
      gVoiceReadIndex  = 0;
    }

  gVoiceID[Index] = VoiceID;

  gVoiceWriteIndex++;
}

uint8_t AUDIO_SetDigitVoice(uint8_t Index, uint16_t Value)
{
  uint16_t Remainder;
  uint8_t  Result;
  uint8_t  Count;

  if (Index == 0)
    {
      gVoiceWriteIndex = 0;
      gVoiceReadIndex  = 0;
    }

  Count     = 0;
  Result    = Value / 1000U;
  Remainder = Value % 1000U;
  if (Remainder < 100U)
    {
      if (Remainder < 10U)
	goto Skip;
    }
  else
    {
      Result = Remainder / 100U;
      gVoiceID[gVoiceWriteIndex++] = (VOICE_ID_t)Result;
      Count++;
      Remainder -= Result * 100U;
    }
  Result = Remainder / 10U;
  gVoiceID[gVoiceWriteIndex++] = (VOICE_ID_t)Result;
  Count++;
  Remainder -= Result * 10U;

 Skip:
  gVoiceID[gVoiceWriteIndex++] = (VOICE_ID_t)Remainder;

  return Count + 1U;
}

void AUDIO_PlayQueuedVoice(void)
{
  uint8_t VoiceID;
  uint8_t Delay;
  bool    Skip;

  Skip = false;

  if (gVoiceReadIndex != gVoiceWriteIndex && gEeprom.VOICE_PROMPT != VOICE_PROMPT_OFF)
    {
      VoiceID = gVoiceID[gVoiceReadIndex];
      if (gEeprom.VOICE_PROMPT == VOICE_PROMPT_CHINESE)
	{
	  if (VoiceID < ARRAY_SIZE(VoiceClipLengthChinese))
	    {
	      Delay = VoiceClipLengthChinese[VoiceID];
	      VoiceID += VOICE_ID_CHI_BASE;
	    }
	  else
	    Skip = true;
	}
      else
	{
	  if (VoiceID < ARRAY_SIZE(VoiceClipLengthEnglish))
	    {
	      Delay = VoiceClipLengthEnglish[VoiceID];
	      VoiceID += VOICE_ID_ENG_BASE;
	    }
	  else
	    Skip = true;
	}

      gVoiceReadIndex++;

      if (!Skip)
	{
	  if (gVoiceReadIndex == gVoiceWriteIndex)
	    Delay += 3;

	  AUDIO_PlayVoice(VoiceID);

	  gCountdownToPlayNextVoice_10ms = Delay;
	  gFlagPlayQueuedVoice           = false;

#ifdef ENABLE_VOX
	  gVoxResumeCountdown = 2000;
#endif

	  return;
	}
    }

  if (FUNCTION_IsRx())
    {
      RADIO_SetModulation(gRxVfo->Modulation); // 1of11
    }

#ifdef ENABLE_FMRADIO
  if (gFmRadioMode)
    BK1080_Mute(false);
#endif

  if (!gEnableSpeaker)
    AUDIO_AudioPathOff();

#ifdef ENABLE_VOX
  gVoxResumeCountdown = 80;
#endif

  gVoiceWriteIndex    = 0;
  gVoiceReadIndex     = 0;
}

#endif
